tensor([[[0.2600, 0.2600],
         [0.2600, 0.2600]],

        [[0.3200, 0.3200],
         [0.3200, 0.3200]],

        [[0.4100, 0.4100],
         [0.4100, 0.4100]]], grad_fn=<RoundBackward1>)